package com.xiongtian.springframework.context.support;

import com.xiongtian.springframework.beans.BeansException;
import com.xiongtian.springframework.beans.factory.config.BeanPostProcessor;
import com.xiongtian.springframework.context.ApplicationContext;
import com.xiongtian.springframework.context.ApplicationContextAware;

/**
 * @Author: xiongtian
 * @CreateTime: 2025-04-02  09:49
 * @Version: 1.0
 * @Description:
 * 由于 ApplicationContext 的获取并不能直接在创建 Bean 时候就可以拿到，所以
 * 需要在 refresh 操作时，把 ApplicationContext 写入到一个包装的
 * BeanPostProcessor 中去，再由AbstractAutowireCapableBeanFactory.applyBeanPostProcessorsBeforeInitialization
 * 方法调用。
 * 所以这里需要实现 BeanPostProcessor 接口，在 Bean 初始化之前，将 ApplicationContext 写入到 Bean 中。
 * 具体实现是在 postProcessBeforeInitialization 方法中实现的。
 * 这里需要注意的是，ApplicationContextAware 是一个标记接口，需要在实现类中实现 setApplicationContext 方法，
 * 才能将 ApplicationContext 写入到 Bean 中。
 */

public class ApplicationContextAwareProcessor implements BeanPostProcessor {

    private final ApplicationContext applicationContext;

    public ApplicationContextAwareProcessor(ApplicationContext applicationContext) {
        this.applicationContext = applicationContext;
    }


    /**
     * 在bean初始化之前执行
     * 如果bean实现了ApplicationContextAware接口，就会调用setApplicationContext方法，将ApplicationContext对象传递给bean
     *
     * @param bean     bean对象
     * @param beanName bean名称
     * @return
     * @throws BeansException
     */
    @Override
    public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
        if (bean instanceof ApplicationContextAware) {
            ((ApplicationContextAware) bean).setApplicationContext(applicationContext);
        }
        return bean;
    }

    @Override
    public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
        return bean;
    }
}
